package com.wc.data_structures.sparsearray;

/**
 * @Author congge
 * @Date 2023/4/18 15:10
 * @description 稀疏数组
 */
public class SparseArray {
    public static void main(String[] args) {
        // 创建一个原始的二维数组 11 * 11
        // 0:表示没有旗子，1 表示 黑子 2 表示白子
        // 初始数组1 2 位置为 1       2 3 位置为 2
        // 棋盘的行列
        int n = 11;
        int[][] array = new int[11][11];
        array[1][2] = 1;
        array[2][3] = 2;
        SparseArray sparseArray = new SparseArray();
        int[][] sparseArray1 = sparseArray.arrayToSparseArray(array);
    }

    /**
     * 将矩阵转换为稀疏矩阵
     *
     * @param chess
     * @return
     */
    public int[][] arrayToSparseArray(int[][] chess) {
        // 表示矩阵为m 行  n 列
        int m = chess.length;
        int n = chess[0].length;
        // 将二维数组转化为稀疏矩阵

        // 1. 遍历二维数组 得到非0的个数
        int sum = 0;
        // 输出原始二维数组
        for (int[] row : chess) {
            for (int num : row) {
//                System.out.printf("%d\t", num);
                if (num != 0) {
                    sum++;
                }
            }
//            System.out.println();
        }

        // 2.构建稀疏矩阵
        int[][] sparseArray = new int[sum + 1][3];
        sparseArray[0][0] = m;
        sparseArray[0][1] = n;
        sparseArray[0][2] = sum;

        // 创建定位器
        int idx = 1;
        // 3.将非零元素存放到稀疏数组中
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (chess[i][j] != 0) {
                    sparseArray[idx][0] = i;
                    sparseArray[idx][1] = j;
                    sparseArray[idx][2] = chess[i][j];
                    idx++;
                }
            }
        }

        return sparseArray;
    }

    /**
     * 将稀疏数组恢复为原始数组
     * @param sparseArray
     * @return
     */
    public int[][] sparseArrayToArray(int[][] sparseArray) {
        // 获取到行列 m  n, 以及非零元素个数
        int m = sparseArray[0][0];
        int n = sparseArray[0][1];
        int sum = sparseArray[0][2];

        // 构建原始数组
        int[][] array = new int[m][n];

        // 创建定位
        int idx = 1;
        int i;
        int j;
        int value;

        // 填充数据
        while (sum-- > 0) {
            i = sparseArray[idx][0];
            j = sparseArray[idx][1];
            value = sparseArray[idx][2];
            array[i][j] = value;
        }
        return array;
    }
}
